python transpose a matrix program